tensor([[[[[0.8200, 0.1700],
           [0.1700, 0.1500]],

          [[0.6000, 0.7100],
           [0.7100, 0.6000]]],


         [[[0.6000, 0.7100],
           [0.7100, 0.6000]],

          [[0.0000, 0.0000],
           [0.0000, 0.0000]]]],



        [[[[0.0000, 0.4300],
           [0.4300, 0.8500]],

          [[0.2300, 0.1700],
           [0.1700, 0.2300]]],


         [[[0.2300, 0.1700],
           [0.1700, 0.2300]],

          [[0.0000, 0.0000],
           [0.0000, 0.0000]]]],



        [[[[0.0000, 0.0000],
           [0.0000, 0.0000]],

          [[0.0000, 0.0000],
           [0.0000, 0.0000]]],


         [[[0.0000, 0.0000],
           [0.0000, 0.0000]],

          [[1.0000, 1.0000],
           [1.0000, 1.0000]]]],



        [[[[0.1800, 0.4000],
           [0.4000, 0.0000]],

          [[0.1600, 0.1200],
           [0.1200, 0.1600]]],


         [[[0.1600, 0.1200],
           [0.1200, 0.1600]],

          [[0.0000, 0.0000],
           [0.0000, 0.0000]]]]], grad_fn=<RoundBackward1>)